// 正确导入 OpenAI 库
const OpenAI = require('openai');
const Router = require('koa-router');
const router = new Router();
const env = require("@/config/config.default")

// 初始化 OpenAI 客户端 - 使用 DeepSeek API
const openai = new OpenAI({
  // apiKey: env.DEEPSEEK_API_KEY, // 从环境变量获取密钥
  apiKey: 'sk-1234567890abcdef1234567890abcdef', // 从环境变量获取密钥
  baseURL: 'https://api.deepseek.com/v1', // DeepSeek API 基础路径
});

// AI 生成文章内容接口
router.post('/deepseek/stream', async (ctx) => {
  try {
    const { prompt } = ctx.request.body;
    if (!prompt) {
      return ctx.body = { code: 400, message: '请提供生成提示词' };
    }

    // 使用正确的方法调用聊天完成 API
    const response = await openai.chat.completions.create({
      model: 'deepseek-chat', // 使用 DeepSeek 模型
      messages: [
        { role: 'user', content: prompt }
      ],
      max_tokens: 1000,
    });

    ctx.body = {
      code: 200,
      data: {
        content: response.choices[0].message.content
      }
    };
  } catch (error) {
    ctx.body = {
      code: 500,
      message: 'AI 内容生成失败: ' + error.message
    };
  }
});

module.exports = router;